import argparse

pair2_parser = argparse.ArgumentParser(description='ArgParser for pair2 train')
pair2_parser.add_argument('exname',type=str,default=None)
pair2_parser.add_argument('savedir', type=str, default=None)
pair2_parser.add_argument('--epochs', type=int, default=80)
pair2_parser.add_argument('--train_iter_num', type=int, default=1000)
pair2_parser.add_argument('--val_iter_num', type=int, default=200)
pair2_parser.add_argument('--eval_freq', type=int, default=2)
pair2_parser.add_argument('--lr', type=float, default=0.1)
pair2_parser.add_argument('--lr_steps', type=list, default=[10, 20, 40])
pair2_parser.add_argument('--lr_delay_rate', type=float, default=0.00005)
pair2_parser.add_argument('--model_arch', type=str, default=None)
pair2_parser.add_argument('--print_freq',type=int,default=50)
pair2_parser.add_argument('--resume_ckpt', type=str, default=None)
pair2_parser.add_argument('--momentum',type=float,default=0.9)
pair2_parser.add_argument('--basestep',type=int,default=0)


class2_parser = argparse.ArgumentParser(description='ArgParser for class2 train')
class2_parser.add_argument('exname',type=str,default=None)
class2_parser.add_argument('savedir', type=str, default=None)
class2_parser.add_argument('--epochs', type=int, default=80)
class2_parser.add_argument('--train_iter_num', type=int, default=1000)
class2_parser.add_argument('--val_iter_num', type=int, default=200)
class2_parser.add_argument('--eval_freq', type=int, default=2)
class2_parser.add_argument('--lr', type=float, default=0.000005)
class2_parser.add_argument('--lr_steps', type=list, default=[10, 20, 40])
class2_parser.add_argument('--lr_delay_rate', type=float, default=0.5)
class2_parser.add_argument('--model_arch', type=str, default=None)
class2_parser.add_argument('--print_freq',type=int,default=50)
class2_parser.add_argument('--resume_ckpt', type=str, default=None)
class2_parser.add_argument('--momentum',type=float,default=0.9)
class2_parser.add_argument('--basestep',type=int,default=0)
